package cool.webstudy.admin.configuration;

import io.swagger.v3.oas.models.Components;
import io.swagger.v3.oas.models.OpenAPI;
import io.swagger.v3.oas.models.info.Info;
import io.swagger.v3.oas.models.security.SecurityRequirement;
import io.swagger.v3.oas.models.security.SecurityScheme;
import lombok.RequiredArgsConstructor;
import org.springdoc.core.customizers.GlobalOpenApiCustomizer;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.HttpHeaders;

/**
 * @author 莫振双
 * @date 2024/7/21 18:09
 * @description: knife4j配置
 */
@Configuration
@RequiredArgsConstructor
public class Knife4jConfig {
    @Value("#{'${security.url_white_list}'.split(',')}")
    private String[] urlWhiteList;
    /**
     * 接口信息
     */
    @Bean
    public OpenAPI openApi() {
        return new OpenAPI()
                .info(new Info()
                        .title("AWD训练平台接口文档")
                        .version("0.0.1")
                )
                // 配置全局鉴权参数-Authorization
                .components(new Components()
                        .addSecuritySchemes(HttpHeaders.AUTHORIZATION,
                                new SecurityScheme()
                                        .name(HttpHeaders.AUTHORIZATION)
                                        .type(SecurityScheme.Type.APIKEY)
                                        .in(SecurityScheme.In.HEADER)
                                        .scheme("Bearer")
                                        .bearerFormat("JWT")
                        )
                );
    }

    /**
     * 全局自定义扩展
     * 在OpenAPI规范中，Operation 是一个表示 API 端点（Endpoint）或操作的对象。
     * 每个路径（Path）对象可以包含一个或多个 Operation 对象，用于描述与该路径相关联的不同 HTTP 方法（例如 GET、POST、PUT 等）。
     */
    @Bean
    public GlobalOpenApiCustomizer globalOpenApiCustomizer() {
        return openApi -> {
            // 全局添加鉴权参数
            if (openApi.getPaths() != null) {
                openApi.getPaths().forEach((s, pathItem) -> {
                    //遍历白名单并构建路径匹配对象数组与请求路径匹配，如果匹配成功则放行
                    for (int i = 0; i < urlWhiteList.length; i++) {
                       if (s.equals(urlWhiteList[i])) {
                           return;
                       }
                    }
                    // 接口添加鉴权参数
                    pathItem.readOperations().forEach(operation ->
                        operation.addSecurityItem(new SecurityRequirement().addList(HttpHeaders.AUTHORIZATION))
                    );
                });
            }
        };
    }
}
